import os

from common import settings
from common.log import logger


def mkdir_classname(tag):
    """
    每个测试类运行时，以类名在截屏目录下创建目录
    :param tag: 测试版本标识
    :return:
    """
    def wrapper(obj):
        if tag.isalnum():
            ver_path = os.path.join(settings.SCREENSHOT_DIR, tag)
            if not os.path.exists(ver_path):
                os.mkdir(ver_path)  # 以测试版本创建一个文件夹

            classname_path = os.path.join(ver_path, obj.__name__)
            if not os.path.exists(classname_path):
                os.mkdir(classname_path)  # 以类名创建一个文件夹
            return obj
        else:
            raise TypeError('测试版本标志只支持英文、数字、英文数字混合！')
    return wrapper

import sys
import logging
def log4gs(obj):
    """
    提供日志打印
    :return:
    """
    obj.log = logger
    return obj

